/**
 * Software License, Version 1.0
 * 
 * Copyright 2003 The Trustees of Indiana University.  All rights reserved.
 * 
 *
 *Redistribution and use in source and binary forms, with or without 
 *modification, are permitted provided that the following conditions are met:
 *
 *1) All redistributions of source code must retain the above copyright notice,
 * the list of authors in the original source code, this list of conditions and
 * the disclaimer listed in this license;
 *2) All redistributions in binary form must reproduce the above copyright 
 * notice, this list of conditions and the disclaimer listed in this license in
 * the documentation and/or other materials provided with the distribution;
 *3) Any documentation included with all redistributions must include the 
 * following acknowledgement:
 *
 *"This product includes software developed by the Community Grids Lab. For 
 * further information contact the Community Grids Lab at 
 * http://communitygrids.iu.edu/."
 *
 * Alternatively, this acknowledgement may appear in the software itself, and 
 * wherever such third-party acknowledgments normally appear.
 * 
 *4) The name Indiana University or Community Grids Lab or NaradaBrokering, 
 * shall not be used to endorse or promote products derived from this software 
 * without prior written permission from Indiana University.  For written 
 * permission, please contact the Advanced Research and Technology Institute 
 * ("ARTI") at 351 West 10th Street, Indianapolis, Indiana 46202.
 *5) Products derived from this software may not be called NaradaBrokering, 
 * nor may Indiana University or Community Grids Lab or NaradaBrokering appear
 * in their name, without prior written permission of ARTI.
 * 
 *
 * Indiana University provides no reassurances that the source code provided 
 * does not infringe the patent or any other intellectual property rights of 
 * any other entity.  Indiana University disclaims any liability to any 
 * recipient for claims brought by any other entity based on infringement of 
 * intellectual property rights or otherwise.  
 *
 *LICENSEE UNDERSTANDS THAT SOFTWARE IS PROVIDED "AS IS" FOR WHICH NO 
 *WARRANTIES AS TO CAPABILITIES OR ACCURACY ARE MADE. INDIANA UNIVERSITY GIVES
 *NO WARRANTIES AND MAKES NO REPRESENTATION THAT SOFTWARE IS FREE OF 
 *INFRINGEMENT OF THIRD PARTY PATENT, COPYRIGHT, OR OTHER PROPRIETARY RIGHTS. 
 *INDIANA UNIVERSITY MAKES NO WARRANTIES THAT SOFTWARE IS FREE FROM "BUGS", 
 *"VIRUSES", "TROJAN HORSES", "TRAP DOORS", "WORMS", OR OTHER HARMFUL CODE.  
 *LICENSEE ASSUMES THE ENTIRE RISK AS TO THE PERFORMANCE OF SOFTWARE AND/OR 
 *ASSOCIATED MATERIALS, AND TO THE PERFORMANCE AND VALIDITY OF INFORMATION 
 *GENERATED USING SOFTWARE.
 */
package cgl.narada.test;

import java.io.BufferedReader;
import java.io.InputStreamReader;
import java.util.Properties;

import javax.jms.JMSException;
import javax.jms.Message;
import javax.jms.MessageListener;
import javax.jms.Session;
import javax.jms.TextMessage;
import javax.jms.Topic;
import javax.jms.TopicConnection;
import javax.jms.TopicConnectionFactory;
import javax.jms.TopicPublisher;
import javax.jms.TopicSession;
import javax.jms.TopicSubscriber;

import cgl.narada.jms.NBJmsInitializer;

/** 
    This program is used to test the unlimited concurrent connection support
    in Narada. This class would be modified to reflect any additional tests
    that need to be performed.
    
    @author Shrideep Pallickara
    $Revision$
    $Date$
*/ 


public class MultipleJmsConnections implements MessageListener {
  
  private String hostname;
  private int portnum;
  private TopicPublisher publisher;
  private TextMessage message;
  private boolean initNeeded = true;
  private TopicPublisher[] publishers;

  public MultipleJmsConnections(String hostName, int portNum, 
				int numConnections) {
    boolean createTopic = true;
    Topic jmsTopic=null;
    hostname = hostName;
    portnum = portNum;
    
    publishers = new TopicPublisher[numConnections];
    Properties props = new Properties();
    /** These properties pertain to setting up a TCP link */
    props.put("hostname", hostname); 
    props.put("portnum", ""+portNum);
    
    int entityId = 10000;
    try {
      for (int i=0; i<numConnections; i++) {
	NBJmsInitializer ini = new NBJmsInitializer(props, "pooltcp", entityId++);
	// Lookup a JMS connection factory
	TopicConnectionFactory conFactory =
	  (TopicConnectionFactory)ini.lookup();
	
	// Create a JMS connection
	TopicConnection connection =
	  conFactory.createTopicConnection("guest","password");
	TopicSession subSession =
	  connection.createTopicSession(false,
					Session.AUTO_ACKNOWLEDGE);
	if (createTopic) {
	  jmsTopic = subSession.createTopic("/streams/chat/private/Text");
	  createTopic=false;
	}
	TopicSubscriber subscriber = 
	  subSession.createSubscriber(jmsTopic);
	Thread.currentThread().sleep(250);
	subscriber.setMessageListener(this);
	publishers[i] = subSession.createPublisher(jmsTopic);

	connection.start();
      }/*end for */
    } catch (JMSException jmsEx) {
      jmsEx.printStackTrace();
    } catch (InterruptedException ie) {
      ie.printStackTrace();
    } /*end try */
  }

  /* Receive message from topic subscriber */
  public void onMessage(Message message) {
    try {
      TextMessage textMessage = (TextMessage) message;
      String text = textMessage.getText();
      System.out.println("JMSUser->" + text);
    } catch(JMSException jmse){ jmse.printStackTrace(); }
  }
  

  protected void writeMessage(String text) throws JMSException {
    
    Topic jmsTopic;
    Properties props = new Properties();
    /** These properties pertain to setting up a TCP link */
    props.put("hostname", hostname); 
    props.put("portnum", ""+portnum);
    
    if (initNeeded) {
      NBJmsInitializer ini = new NBJmsInitializer(props, "pooltcp");
      // Lookup a JMS connection factory
      TopicConnectionFactory conFactory =
	(TopicConnectionFactory)ini.lookup();
      
      // Create a JMS connection
      TopicConnection connection =
	conFactory.createTopicConnection("guest","password");
      TopicSession pubSession =
	connection.createTopicSession(false,
				      Session.AUTO_ACKNOWLEDGE);
      jmsTopic = pubSession.createTopic("/streams/chat/private/Text");
      publisher = 
	pubSession.createPublisher(jmsTopic);
      message = pubSession.createTextMessage();
      initNeeded = false;
    } /*end init needed */
   
    message.setText("Publisher : "+text);
    publisher.publish(message);
  }
  

  protected void writeMultipleMessages(String text) throws JMSException {
    if (initNeeded) {
      writeMessage(text);
    }
    for (int i=0; i < publishers.length; i++) {
       message.setText("pub( "+ i + ")" + text);
       publishers[i].publish(message);
    }

  }



  public static void main(String[] args) {
    if (args.length !=3) {
      System.out.println("Usage> java cgl.narada.test.MultipleJmsConnections " +
			 "<hostName> <portNum> <numOfConnections>");
      System.exit(0);
    }
    
    String hostname = args[0];
    int portNum = Integer.parseInt(args[1]);
    int numConnections = Integer.parseInt(args[2]);
    
    MultipleJmsConnections mJmsConns = 
      new MultipleJmsConnections(hostname, portNum, numConnections);
   
    try {
      BufferedReader commandLine = 
	new BufferedReader(new InputStreamReader(System.in));
      while(true){
	String s = commandLine.readLine();
	if(s.equalsIgnoreCase("exit")){
	  System.exit(0);// exit program
	}
	if (s.startsWith("wm")) {
	  mJmsConns.writeMultipleMessages(s);
	}

	mJmsConns.writeMessage(s);
      }
    } catch (Exception e) {
      e.printStackTrace();
    }
  }/*end main*/

}
